package com.liziba.trend.utils;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.BoundListOperations;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;
import org.springframework.util.Assert;
import org.springframework.util.CollectionUtils;

import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.TimeUnit;

/**
 * @ProjectName: trend
 * @Package: com.liziba.trend.utils
 * @ClassName: RedisUtil
 * @Author: liziba
 * @Description: redis工具类
 * @Date: 2021/2/2 11:52
 * @Version: 1.0
 */

//@Component
public class RedisUtils {

    @Autowired
    private RedisTemplate<String, Object> redisTemplate;

    public RedisUtils(RedisTemplate<String, Object> redisTemplate) {
        this.redisTemplate = redisTemplate;
    }

    /**
     * 指定缓存失效时间
     * @param key       键
     * @param time      失效时间
     * @return
     */
    public boolean expire(String key, long time) {
        try {
            if(time > 0) {
                redisTemplate.expire(key, time, TimeUnit.SECONDS);
            }
            return true;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }

    /**
     * 获取缓存失效时间
     * @param key       键
     * @return
     */
    public long getExpire(String key) {
        try {
            return redisTemplate.getExpire(key);
        } catch (Exception e){
            e.printStackTrace();
        }
        return 0;
    }

    /**
     * 判断key是否存在
     * @param key        键
     * @return
     */
    public boolean hasKey(String key) {
        try {
           return redisTemplate.hasKey(key);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }

    /**
     * 删除缓存
     * @param key       键 -- 支持一次移除多个缓存key
     */
    public void delete(String ... key) {
        if(key != null && key.length > 0) {
            if(key.length == 1) {
                redisTemplate.delete(key[0]);
            } else {
                redisTemplate.delete(CollectionUtils.arrayToList(key));
            }
        }
    }


    // ==========================================String 数据类型=========================================

    /**
     * 普通缓存获取
     * @param key
     * @return
     */
    public Object get(String key) {
        return key == null ? null : redisTemplate.opsForValue().get(key);
    }

    /**
     * 普通缓存放入
     * @param key
     * @param value
     * @return
     */
    public boolean set(String key, Object value) {
        try {
            redisTemplate.opsForValue().set(key, value);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }

    /**
     * 普通缓存放入&设置过期时间
     * @param key           键
     * @param value         值
     * @param expire        国企时间
     * @return
     */
    public boolean set(String key, Object value, long expire) {
        try {
            redisTemplate.opsForValue().set(key, value, expire, TimeUnit.SECONDS);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }

    /**
     * 递增
     * @param key       键
     * @param delta     递增因子
     * @return
     */
    public long increment(String key, long delta) {
        Assert.isTrue((delta < 0), "递增因子必须大于0！");
        return redisTemplate.opsForValue().increment(key, delta);
    }

    /**
     * 递减
     * @param key           键
     * @param delta         递减因子
     * @return
     */
    public long decrement(String key, long delta) {
        Assert.isTrue((delta < 0), "递减因子必须大于0！");
        return redisTemplate.opsForValue().decrement(key, delta);
    }

    // =====================================Hash================================

    /**
     * HashGet
     * @param key    键 不能为null
     * @param item   项 不能为null
     * @return       值
     */
    public Object hget(String key, String item) {
        Assert.isNull(key, "键 不能为null");
        Assert.isNull(item, "项 不能为null");
        return redisTemplate.opsForHash().get(key, item);
    }

    /**
     * 获取key对应的所有键值对
     * @param key       键 不能为null
     * @return          对应的多个键值
     */
    public Map<Object, Object> hmget(String key) {
        Assert.isNull(key, "键 不能为null");
        return redisTemplate.opsForHash().entries(key);
    }

    /**
     * HashSet
     * @param key       键
     * @param map       对应的多个键值对
     * @return
     */
    public boolean hmset(String key, Map<String, Object> map) {
        try {
            redisTemplate.opsForHash().putAll(key, map);
            return true;
        } catch (Exception e){
            e.printStackTrace();
        }
        return false;
    }

    /**
     * HashSet
     * @param key       键
     * @param map       对应的多个键值对
     * @param time     过期时间
     * @return
     */
    public boolean hmset(String key, Map<String, Object> map, long time) {
        try {
            redisTemplate.opsForHash().putAll(key, map);
            if(time > 0) {
                expire(key, time);
            }
            return true;
        } catch (Exception e){
            e.printStackTrace();
        }
        return false;
    }

    /**
     * 向一张hash表中放入数据如果不存在则创建
     * @param key           键
     * @param item          项
     * @param value         值
     * @return
     */
    public boolean hset(String key, Object item, Object value) {
        try {
            redisTemplate.opsForHash().put(key, item, value);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }

    /**
     * 向一张hash表中放入数据如果不存在则创建
     * @param key           键
     * @param item          项
     * @param value         值
     * @param time          过期时间    如果已存在的hash表有时间,这里将会替换原有的时间
     * @return
     */
    public boolean hset(String key, Object item, Object value, long time) {
        try {
            redisTemplate.opsForHash().put(key, item, value);
            if(time > 0) {
                expire(key, time);
            }
            return true;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }

    /**
     * 删除hash表中的值
     * @param key       键
     * @param item      项
     */
    public void hdel(String key, Object ... item) {
        Assert.isNull(key, "键 不能为null");
        Assert.isNull(item, "项 不能为null");
        redisTemplate.opsForHash().delete(key, item);
    }


    /**
     * 判断hash表中是否存在该项的值
     * @param key       键
     * @param item      项
     * @return
     */
    public boolean hHasKey(String key, Object item) {
        Assert.isNull(key, "键 不能为null");
        Assert.isNull(item, "项 不能为null");
       return redisTemplate.opsForHash().hasKey(key, item);
    }


    /**
     * hash递增，如果不存在，就会创建一个，并返回新的值返回
     * @param key       键
     * @param item      项
     * @param by        递增值
     * @return
     */
    public double hincr(String key, Object item, double by) {
        Assert.isTrue((by < 0), "递增因子必须大于0！");
        return redisTemplate.opsForHash().increment(key, item, by);
    }

    /**
     * hash递减，如果不存在，就会创建一个，并返回新的值返回
     * @param key       键
     * @param item      项
     * @param by        递减值
     * @return
     */
    public double hdeci(String key, Object item, double by) {
        Assert.isTrue((by < 0), "递减因子必须大于0！");
        return redisTemplate.opsForHash().increment(key, item, -by);
    }


    /**
     * hash递增，如果不存在，就会创建一个，并返回新的值返回
     * @param key       键
     * @param item      项
     * @param by        递增值
     * @return
     */
    public long hincr(String key, Object item, long by) {
        Assert.isTrue((by < 0), "递增因子必须大于0！");
        return redisTemplate.opsForHash().increment(key, item, by);
    }

    /**
     * hash递减，如果不存在，就会创建一个，并返回新的值返回
     * @param key       键
     * @param item      项
     * @param by        递减值
     * @return
     */
    public long hdeci(String key, Object item, long by) {
        Assert.isTrue((by < 0), "递减因子必须大于0！");
        return redisTemplate.opsForHash().increment(key, item, -by);
    }

    // =================================== set ====================================

    /**
     * 根据key获取set中所有的值
     * @param key       键
     * @return
     */
    public Set<Object> sGet(String key) {
        Assert.isNull(key, "键 不能为null");
        return redisTemplate.opsForSet().members(key);
    }

    /**
     * 根据value从一个set中查询,是否存在
     * @param key           键
     * @param value         值
     * @return
     */
    public boolean sHasKey(String key, Object value) {
        Assert.isNull(key, "键 不能为null");
        return redisTemplate.opsForSet().isMember(key, value);
    }

    /**
     * 将数据放入set缓存
     * @param key           键
     * @param values        值
     * @return
     */
    public long sSet(String key, Object ... values) {
        Assert.isNull(key, "键 不能为null");
        Assert.isNull(values, "值 不能为null");
        try {
            return redisTemplate.opsForSet().add(key, values);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return 0;
    }

    /**
     * 将数据放入set缓存&设置国企时间
     * @param key           键
     * @param time          过期时间
     * @param values        值
     * @return
     */
    public long sSet(String key, long time, Object ... values) {
        Assert.isNull(key, "键 不能为null");
        Assert.isNull(values, "值 不能为null");
        try {
            Long size = redisTemplate.opsForSet().add(key, values);
            if(time > 0) {
                expire(key, time);
            }
            return size;
        } catch (Exception e){
            e.printStackTrace();
        }
        return 0;
    }

    /**
     * 根据key获取set缓存的长度
     * @param key           键
     * @return
     */
    public long sGetSetSize(String key) {
        Assert.isNull(key, "键 不能为null");
        try {
            return redisTemplate.opsForSet().size(key);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return 0;
    }

    /**
     * 根据key移除set缓存中的数据
     * @param key           键
     * @param values        值
     * @return
     */
    public long sRemove(String key, Object ... values) {
        Assert.isNull(key, "键 不能为null");
        Assert.isNull(values, "值 不能为null");
        try {
            return redisTemplate.opsForSet().remove(key, values);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return 0;
    }

    // ==================================== list ===========================

    /**
     * 获取list缓存中的内容
     * @param key               键
     * @param start             起始位置
     * @param end               结束位置
     * @return
     */
    public List<Object> lGet(String key, long start, long end) {
        Assert.isNull(key, "键 不能为null");
        try {
            return redisTemplate.opsForList().range(key, start, end);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

    /**
     * 获取list缓存的长度
     * @param key           键
     * @return
     */
    public long lGetListSize(String key) {
        Assert.isNull(key, "键 不能为null");
        try {
           return redisTemplate.opsForList().size(key);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return 0;
    }

    /**
     * 根据索引获取list缓存中的数据
     * @param key           键
     * @param index         索引   index >= 0  leftPop 表头 index <= 0 rightPop 表尾
     * @return
     */
    public Object lGetIndex(String key, long index) {
        Assert.isNull(key, "键 不能为null");
        try {
            return redisTemplate.opsForList().index(key, index);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

    /**
     * list放入缓存  rightPush
     * @param key           键
     * @param value         值
     * @return
     */
    public boolean lSet(String key, Object value) {
        Assert.isNull(key, "键 不能为null");
        try {
            redisTemplate.opsForList().rightPush(key, value);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }

    /**
     * list放入缓存  rightPush 并设置过期时间
     * @param key           键
     * @param time          过期时间
     * @param value         值
     * @return
     */
    public boolean lSet(String key, Object value, long time) {
        Assert.isNull(key, "键 不能为null");
        try {
            redisTemplate.opsForList().rightPush(key, value);
            if (time > 0) {
                expire(key, time);
            }
            return true;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }

    /**
     * list放入缓存  rightPush
     * @param key           键
     * @param value         值
     * @return
     */
    public boolean lSet(String key, List<Object> value) {
        Assert.isNull(key, "键 不能为null");
        try {
            redisTemplate.opsForList().rightPushAll(key, value);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }

    /**
     * list放入缓存  rightPush 并设置过期时间
     * @param key           键
     * @param time          过期时间
     * @param value         值
     * @return
     */
    public boolean lSet(String key, List<Object> value, long time) {
        Assert.isNull(key, "键 不能为null");
        try {
            redisTemplate.opsForList().rightPushAll(key, value);
            if (time > 0) {
                expire(key, time);
            }
            return true;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }

    /**
     * 根据索引修改list中的某条数据
     * @param key           键
     * @param index         索引
     * @param value         值
     * @return
     */
    public boolean lUpdateIndex (String key, long index, Object value) {
        Assert.isNull(key, "键 不能为null");
        try {
            redisTemplate.opsForList().set(key, index, value);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }

    /**
     *  移除N个值为value
     * @param key           键
     * @param count         移除个数
     * @param value         值
     * @return
     */
    public boolean lRemove(String key, long count, Object value) {
        Assert.isNull(key, "键 不能为null");
        Assert.isTrue((count <= 0), "移除个数必须大于0！");
        try {
            redisTemplate.opsForList().remove(key, count, value);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }

    /**
     * 模糊查询获取key值
     * @param pattern
     * @return
     */
    public Set keys(String pattern) {
        Assert.isNull(pattern, "pattern 不能为null");
        try {
            return redisTemplate.keys(pattern);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }


    /**
     * 使用redis消息队列
     * @param channel
     * @param message
     */
    public void convertAndSend(String channel, Object message) {
        Assert.isNull(channel, "channel 不能为null");
        try {
            redisTemplate.convertAndSend(channel, message);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    // ==================== boundListOps ===================

    /**
     * 根据起始结束序号遍历Redis中的list
     * @param key           键
     * @param start         起始位置
     * @param end           结束位置
     * @return
     */
    public List<Object> rangeList(String key, long start, long end) {
        Assert.isNull(key, "key 不能为null");
        Assert.isTrue((start < 0), "start 必须大于等于0");
        Assert.isTrue((end < 0), "end 必须大于等于0");
        try {
            BoundListOperations<String, Object> stringObjectBoundListOperations = redisTemplate.boundListOps(key);
            return stringObjectBoundListOperations.range(start, end);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

    /**
     * 弹出右边的值 --- 并且移除这个值
     * @param key
     * @return
     */
    public Object rightPop(String key) {
        Assert.isNull(key, "key 不能为null");
        try {
            BoundListOperations<String, Object> stringObjectBoundListOperations = redisTemplate.boundListOps(key);
            return stringObjectBoundListOperations.rightPop();
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

}
